import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D  

LAYER_3_NEURONS = 1
model_to_draw3D = Model(inputs=model3D.input,
                        outputs=model3D.layers[LAYER_3_NEURONS].output)
int_out = model_to_draw3D.predict(X_test)
print(int_out.shape)
